package com.qsl.ssyx.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

@Component
public class MQProducerAckConfig implements RabbitTemplate.ReturnCallback, RabbitTemplate.ConfirmCallback {

    //  我们发送消息使用的是 private RabbitTemplate rabbitTemplate; 对象
    //  如果不做设置的话 当前的rabbitTemplate 与当前的配置类没有任何关系！
    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * `@PostConstruct` 注解:表示在Bean初始化完成之后会被自动调用。
     * rabbitTemplate.setReturnCallback(this)` 方法设置了一个回调函数 `this` 来处理消息发送中的返回信息。当消息无法被路由到对应的队列时，会触发这个回调函数，可以在回调函数中进行相应的处理。
     * 接下来，使用 `rabbitTemplate.setConfirmCallback(this)` 方法设置了另一个回调函数 `this` 来处理消息发送的确认机制。当消息成功发送到交换机时，会触发这个回调函数，可以在回调函数中进行相应的处理。
     * 通过设置这两个回调函数，可以在消息发送过程中获得返回信息和确认信息，从而监控和处理消息发送的结果。
     * 需要注意的是，这里的 `rabbitTemplate` 是一个消息操作模板，用于发送和接收消息。通过设置回调函数，可以在发送消息时监听返回结果和确认结果，并进行相应的处理。
     */
    // 设置，表示修饰一个非静态的void方法，在服务器加载Servlet的时候运行。并且只执行一次。
    @PostConstruct
    public void init() {
        rabbitTemplate.setReturnCallback(this);
        rabbitTemplate.setConfirmCallback(this);
    }

    /**
     * 判断消息是否正确发送到了交换机上
     *
     * @param correlationData 消息的载体
     * @param ack             判断消息是否发送到交换机上
     * @param cause           发送失败原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        if (ack) {
            System.out.println("消息发送成功");
        } else {
            System.out.println("消息发送失败：" + cause);
        }
    }

    /**
     * 消息如果没有发送到队列中，执行此方法
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        System.out.println("消息主体：" + new String(message.getBody()));
        System.out.println("应答码：" + replyCode);
        System.out.println("描述：" + replyText);
        System.out.println("交换器 exchange : " + exchange);
        System.out.println("路由键 routing : " + routingKey);
    }
}
